package com.example.consumer;

import com.example.entity.Order;
import com.rabbitmq.client.AMQP;
import com.rabbitmq.client.Channel;
import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.RabbitListener;
import org.springframework.stereotype.Service;

import java.io.IOException;
import java.util.HashMap;
import java.util.Map;

@Service
@Slf4j
public class OrderDlxConsumer {

    @RabbitListener(queues = "order.dlx.queue")
    public void handleOrder(Order order, Channel channel, Message message) throws IOException {
        // 正确地从 Message 对象中获取 deliveryTag
        long deliveryTag = message.getMessageProperties().getDeliveryTag();
        try {
            // 业务处理
            log.info("order.dlx.queue处理订单：{}", order);
            processOrder(order);
            // 手动确认消息（deliveryTag：消息唯一标识；multiple：是否批量确认）
            channel.basicAck(message.getMessageProperties().getDeliveryTag(), false);
        } catch (Exception e) {
            log.error("处理订单失败: {}", order, e);

            // 获取重试次数
            Map<String, Object> headers = message.getMessageProperties().getHeaders();
            Integer retryCount = (Integer) headers.getOrDefault("x-retry-count", 0);
            log.info("当前重试次数：{}", retryCount);
            try {
                // 消息处理失败，根据业务需求决定是否重新入队
                if (retryCount < 5) { // 最大重试3次
                    // 重新入队逻辑
                    AMQP.BasicProperties.Builder builder = new AMQP.BasicProperties.Builder();
                    Map<String, Object> newHeaders = new HashMap<>(headers);
                    newHeaders.put("x-retry-count", retryCount + 1);
                    builder.headers(newHeaders);

                    // 重新发布消息
                    channel.basicPublish("order.dlx.exchange", "order.dlx", builder.build(), message.getBody());
                    // 确认原消息
                    channel.basicAck(deliveryTag, false);
                } else {
                    // 达到最大重试次数，拒绝消息并进入死信队列
                    channel.basicNack(deliveryTag, false, false);
                    log.warn("Message rejected after {} retries: {}", retryCount, order);
                }
            } catch (IOException ioException) {
                log.error("处理消息重试失败", ioException);
                try {
                    // 最后的兜底处理
                    channel.basicNack(deliveryTag, false, false);
                } catch (IOException ex) {
                    log.error("最终确认消息失败", ex);
                }
            }
        }
    }

    /**
     * 处理订单业务逻辑
     * @param order 订单对象
     */
    private void processOrder(Order order) {
        // 模拟订单处理逻辑
        log.info("死信开始处理订单，订单号：{}", order.getId());

        // 这里添加实际的业务处理逻辑
        // 例如：更新数据库、调用其他服务、发送通知等

        // 模拟处理时间
        try {
            Thread.sleep(1000);
        } catch (InterruptedException e) {
            Thread.currentThread().interrupt();
            throw new RuntimeException("订单处理被中断", e);
        }

        // 模拟偶尔的处理失败
        if (order.getUserId() != null && order.getUserId().contains("ERROR")) {
            throw new RuntimeException("模拟订单处理失败");
        }

        log.info("订单处理完成，订单号：{}", order.getUserId());
    }

    /**
     * 判断异常是否可重试
     * @param e 异常对象
     * @return 是否可重试
     */
    /**
     * 判断异常是否可重试（增强版）
     * @param e 异常对象
     * @param message 消息对象，用于获取重试次数等信息
     * @return 是否可重试
     */
    private boolean isRetryable(Exception e, Message message) {
        // 获取重试次数
        Integer retryCount = (Integer) message.getMessageProperties().getHeaders().get("retry-count");
        log.info("当前重试次数：{}", retryCount);
        if (retryCount == null) {
            retryCount = 0;
        }

        // 最多重试3次
        if (retryCount >= 3) {
            return false;
        }

        // 其他重试逻辑...
        return isRetryable(e);
    }

    /**
     * 判断异常是否可重试
     * @param e 异常对象
     * @return 是否可重试
     */
    private boolean isRetryable(Exception e) {
        // 根据业务需求判断哪些异常需要重试
        // 例如：网络异常、数据库连接异常等临时性错误可以重试
        // 业务逻辑错误等永久性错误不应重试

        String message = e.getMessage();
        if (message == null) {
            return true; // 未知异常默认重试
        }

        // 永久性错误不重试
        if (message.contains("订单不存在") ||
                message.contains("用户不存在") ||
                message.contains("参数错误")) {
            return false;
        }

        // 临时性错误可以重试
        if (message.contains("网络超时") ||
                message.contains("数据库连接失败") ||
                message.contains("服务不可用")) {
            return true;
        }

        // 默认情况下，允许重试几次后进入死信队列
        return true;
    }
}